understand 2D 3D nD array in python numpy